/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */

package unc.pds.servlets;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.net.URLDecoder;
import java.net.URLEncoder;
import java.util.Iterator;
import java.util.List;
import javax.naming.InitialContext;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileItemFactory;
import org.apache.commons.fileupload.disk.DiskFileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import unc.pds.data.Consts;
import unc.pds.filemodel.FileRemote;
import unc.pds.filemodel.FolderRemote;
import unc.pds.service.ServiceBeanRemote;
import unc.pds.service.ServiceBeanRemoteHome;
import unc.pds.util.Arch;
import unc.pds.util.ArchImpl;

/**
 *
 * @author Mikhail
 */
public class UploadServlet extends HttpServlet {
   
    /** 
     * Processes requests for both HTTP <code>GET</code> and <code>POST</code> methods.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    protected void processRequest(HttpServletRequest request, HttpServletResponse response)
    throws ServletException, IOException {
        response.setContentType("text/html;charset=UTF-8");
        request.setCharacterEncoding("UTF-8");
        PrintWriter out = response.getWriter();
        try {
            ServiceBeanRemote service = ((ServiceBeanRemoteHome)(new InitialContext()).lookup("ejb/ServiceBean")).create((Long)request.getSession().getAttribute("sid"));
            FileItemFactory factory = new DiskFileItemFactory();
            ServletFileUpload upload = new ServletFileUpload(factory);
            List /* FileItem */ items = upload.parseRequest(request);
            Iterator iter = items.iterator();
            String id="";
            while (iter.hasNext()) {
                FileItem item = (FileItem)iter.next();
                if (item.getFieldName().equals("objid")) {
                    id = item.getString();                 
                } else if (item.getFieldName().equals("filename")) {
                    FolderRemote folder = (FolderRemote)service.getByKey(new Long(Long.parseLong(id)));
                    Arch doc = new ArchImpl("root", null);
                    doc.addChild(new ArchImpl(Consts.ATTR__FILE_NAME, item.getName()));
                    doc.addChild(new ArchImpl(Consts.ATTR__FILE_AUTHOR, service.getUser().getKey().toString()));
                    FileRemote file = folder.addFile(doc);
                    File f = new File("files");
                    f.mkdir();
                    f = new File("files\\"+file.getKey());
                    f.createNewFile();
                    item.write(f);
                    response.sendRedirect("index.jsp?id="+id);
                }
            }

        } catch (Exception e) {
            out.print(e.getLocalizedMessage());
        } finally { 
            out.close();
        }
    } 

    // <editor-fold defaultstate="collapsed" desc="HttpServlet methods. Click on the + sign on the left to edit the code.">
    /** 
     * Handles the HTTP <code>GET</code> method.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    protected void doGet(HttpServletRequest request, HttpServletResponse response)
    throws ServletException, IOException {
        processRequest(request, response);
    } 

    /** 
     * Handles the HTTP <code>POST</code> method.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
    throws ServletException, IOException {
        processRequest(request, response);
    }

    /** 
     * Returns a short description of the servlet.
     * @return a String containing servlet description
     */
    public String getServletInfo() {
        return "Short description";
    }// </editor-fold>

}
